class GetConfig:
    def __init__(self) -> None:
        
        self.project_name = 'clip-reid'
        self.use_cuda     = True
        self.seed         = 123


        self.train_batch_size = 180
        self.test_batch_size  = 1


        self.train_path   = ''
        self.train_dir    = '/home/dev/luotianhang/work_item/data_item/reid/Market-1501-v15.09.15'
        self.query_path   = ''
        self.query_dir    = '/mnt/c/Users/luotianhang/work_item/data_item/reid/Market-1501-v15.09.15'
        self.gallery_path = ''
        self.gallery_dir  = '/mnt/c/Users/luotianhang/work_item/data_item/reid/Market-1501-v15.09.15'


        self.train_dir_market1501_dir    = '/home/dev/luotianhang/work_item/data_item/reid/Market-1501-v15.09.15'
        self.train_dir_dukemtmc_dir      = '../../data_item/reid/DukeMTMC-reID'
        self.train_dir_msmt_dir          = '../../data_item/reid/msmt17'

        self.resume            = False
        self.pretrained_weight = ''
        self.save_path         = './weights'

        self.lr     = 1e-2
        self.center_lr = 1e-2
        self.epochs = 100

        self.accumulation_steps = 1

        self.prompt_learner_pretrain_weight = './weights/clip-reid/prompt_learner/152.pth'
        self.image_encoder_pretrain_weight  = './weights/image_encoder_best.pth'
